Advent of Code Python